import torch


print(torch.__version__)          # 查看 PyTorch 版本
print(torch.cuda.is_available())  # 应输出 True
print(torch.version.cuda)         # 应显示 CUDA 版本

